package com.ztc.test;

import org.apache.poi.xwpf.usermodel.ParagraphAlignment;
import org.apache.poi.xwpf.usermodel.XWPFDocument;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.net.URLEncoder;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.xwpf.usermodel.XWPFParagraph;
import org.apache.poi.xwpf.usermodel.XWPFRun;
import org.apache.xmlbeans.XmlException;
import org.apache.xmlbeans.XmlToken;
import org.openxmlformats.schemas.drawingml.x2006.main.CTNonVisualDrawingProps;
import org.openxmlformats.schemas.drawingml.x2006.main.CTPositiveSize2D;
import org.openxmlformats.schemas.drawingml.x2006.wordprocessingDrawing.CTInline;

/**
 * @author dongqianggao
 * @version 2017-12-18 18:08
 */
public class CustomXWPFDocument extends XWPFDocument {

	public static String picType = "jpg";

	public CustomXWPFDocument(InputStream in) throws IOException {
		super(in);
	}

	/**
	 *
	 */
	public CustomXWPFDocument() {
		super();
		// TODO Auto-generated constructor stub
	}

	/**
	 * @param pkg
	 * @throws IOException
	 */
	public CustomXWPFDocument(OPCPackage pkg) throws IOException {
		super(pkg);
		// TODO Auto-generated constructor stub
	} // picAttch 图片后面追加的字符串 可以是空格

	public void createPicture(XWPFParagraph paragraph, int id, int width, int height,
			String picAttch) {
		final int EMU = 9525;
		width *= EMU;
		height *= EMU;
		String blipId = getAllPictures().get(id).getPackageRelationship().getId();

		CTInline inline = paragraph.createRun().getCTR().addNewDrawing().addNewInline();
		paragraph.createRun().setText(picAttch);
		String picXml = ""
				+ "<a:graphic xmlns:a=\"http://schemas.openxmlformats.org/drawingml/2006/main\">"
				+ "   <a:graphicData uri=\"http://schemas.openxmlformats.org/drawingml/2006/picture\">"
				+ "      <pic:pic xmlns:pic=\"http://schemas.openxmlformats.org/drawingml/2006/picture\">"
				+ "         <pic:nvPicPr>" + "            <pic:cNvPr id=\"" + id
				+ "\" name=\"Generated\"/>" + "            <pic:cNvPicPr/>"
				+ "         </pic:nvPicPr>" + "         <pic:blipFill>"
				+ "            <a:blip r:embed=\"" + blipId
				+ "\" xmlns:r=\"http://schemas.openxmlformats.org/officeDocument/2006/relationships\"/>"
				+ "            <a:stretch>" + "               <a:fillRect/>"
				+ "            </a:stretch>" + "         </pic:blipFill>" + "         <pic:spPr>"
				+ "            <a:xfrm>" + "               <a:off x=\"0\" y=\"0\"/>"
				+ "               <a:ext cx=\"" + width + "\" cy=\"" + height + "\"/>"
				+ "            </a:xfrm>" + "            <a:prstGeom prst=\"rect\">"
				+ "               <a:avLst/>" + "            </a:prstGeom>" + "         </pic:spPr>"
				+ "      </pic:pic>" + "   </a:graphicData>" + "</a:graphic>";

		// CTGraphicalObjectData graphicData =
		inline.addNewGraphic().addNewGraphicData();
		XmlToken xmlToken = null;
		try {
			xmlToken = XmlToken.Factory.parse(picXml);
		} catch (XmlException xe) {
			xe.printStackTrace();
		}
		inline.set(xmlToken);
		// graphicData.set(xmlToken);

		inline.setDistT(0);
		inline.setDistB(0);
		inline.setDistL(0);
		inline.setDistR(0);

		CTPositiveSize2D extent = inline.addNewExtent();
		extent.setCx(width);
		extent.setCy(height);

		CTNonVisualDrawingProps docPr = inline.addNewDocPr();
		docPr.setId(id);
		docPr.setName("图片" + id);
		docPr.setDescr("");
	}

	public static void expWord(String path) {
		try {
			CustomXWPFDocument doc = new CustomXWPFDocument(); // 创建文档实体

			XWPFParagraph title = doc.createParagraph(); // 创建一个段落
			title.setAlignment(ParagraphAlignment.CENTER); // 设置段落的位置
			XWPFRun r1 = title.createRun(); // 设置相同样式的文本
			r1.setBold(true); // 设置字体是否加粗
			r1.setFontFamily("宋体"); // 设置字体
			r1.setText("退费审批表"); // 添加文字内容
			r1.setFontSize(16); // 设置字体大小
			// 插入图片文件，同样需要新建一个段落实体
			XWPFParagraph pic = doc.createParagraph();
			pic.setAlignment(ParagraphAlignment.CENTER);
			doc.addPictureData(
					new FileInputStream("C:\\Users\\Administrator\\Desktop\\网络通信协议.jpg"),
					XWPFDocument.PICTURE_TYPE_JPEG);
			doc.createPicture(pic, doc.getAllPictures().size() - 1, 97, 47, "");
			
			// 通过respons输出提示框下载文件，要注意的是如果文件名称中有文字，需要对文件名称进行URLEncoder编码
			String fileName = "退费审批表——1.docx";
			FileOutputStream out = new FileOutputStream(new File(path + fileName));
			doc.write(out);
			out.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
	
	public static void main(String[] args) {
		CustomXWPFDocument.expWord("C:\\Users\\Administrator\\Desktop\\");
	}
}